from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage, SystemMessage

# 初始化ChatOpenAI客户端，连接到GLM-4 API
chat_model = ChatOpenAI(
    api_key="4f2ffee630503342d89d54c5b2f470d8.y2cJ1ENSQiVEWaZO",
    base_url="https://open.bigmodel.cn/api/paas/v4/",
    model="glm-4-flash-250414",
    temperature=0.9,
    top_p=0.7
)

# 准备消息列表
messages = [
    SystemMessage(content="你是一个聪明且富有创造力的小说作家"),
    HumanMessage(content="请你作为童话故事大王，写一篇短篇童话故事。")
]

# 发送请求并获取响应
response = chat_model.invoke(messages)

# 打印结果
print(response.content)